import time
import unittest

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys


class ChromeTestCase(unittest.TestCase):
    """
    这是一个 Chrome 的 测试集
    """
    driver = None

    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Firefox()

    def test_firefox(self):
        self.driver.get("https://www.baidu.com")

        #  想要使用 Xpath 的方式 来定位 百度首页中的文本框
        #  //input[@id='kw']
        # 已经过时 的写法
        # kw = self.driver.find_element_by_xpath("//input[@id='kw']")
        # 向 已经 定位到 的 文本框中 写入 数据
        # kw.send_keys("hanamaki")

        # 使用快捷键  alt + enter
        kw = self.driver.find_element(By.XPATH, "//input[@id='kw']")
        kw.send_keys("hanamaki", Keys.RETURN)
        #
        # self.driver.find_element(By.XPATH, "//input[@id='su']").click()

        # 强制等待 5s
        time.sleep(5)

    @classmethod
    def tearDownClass(cls):
        # 只会 关闭 打开的浏览器中的  一个 标签页
        cls.driver.close()

        # 关闭 打开的浏览器中的  所有 标签页
        # cls.driver.quit()


if __name__ == '__main__':
    unittest.main()
